from selenium.webdriver.common.by import By

from page_object.pages.BasePage import BasePage
from page_object.pages.ReleasePage import ReleasePage


class ReleaseNewsRedPage(BasePage):
    #发动态的发红包页面

    _et_redpackage = (By.ID, 'et_redpackage')
    _et_money_real = (By.ID, 'et_money_real')  #实发金额
    _remark = (By.ID, 'et_content')    #备注
    _tv_redpackage = (By.ID, 'tv_redpackage') #发红包按钮

    def send_coin_red(self, num, money, remark):
        '''
        发送金币红包，因为页面进来就是金币红包页面，所以不动点击
        :param num: 红包数量
        :param money: 红包金额
        :param remark: 备注
        :return:
        '''
        self.find(self._et_redpackage).send_keys(num)
        self.find(self._et_money_real).send_keys(money)
        self.find(self._remark).send_keys(remark)
        self.find(self._tv_redpackage).click()
        return ReleasePage()

    def send_Diamonds_red(self, num, money, remark):
        '''
        发送钻石红包，因为页面进来就是金币红包页面，所以不动点击
        :param num: 红包数量
        :param money: 红包金额
        :param remark: 备注
        :return:
        '''
        self.find_by_xpath('钻石红包').click()
        self.find(self._et_redpackage).send_keys(num)
        self.find(self._et_money_real).send_keys(money)
        self.find(self._remark).send_keys(remark)
        self.find(self._tv_redpackage).click()
        return ReleasePage()



